__author__ = 'Administrator'

from selenium.webdriver.common.by import By

from pages.MainPage import MainPage


class MovePopUpPageLocators(object):
    MOVE_TO_FOLDER_MENU = (By.ID, "divMoveToFolder")
    MOVE_TO_INBOX = (By.ID, "divTrNdO")
    MOVE_BUTTON = (By.ID, "btn1")


class MovePopUpPage(MainPage):
    def click_move_to_folder_menu(self):
        self.find_element(MovePopUpPageLocators.MOVE_TO_FOLDER_MENU, 0).click()

    def click_move_to_inbox(self):
        self.find_element(MovePopUpPageLocators.MOVE_TO_INBOX, 19).click()

    def click_move_button(self):
        self.find_element(MovePopUpPageLocators.MOVE_BUTTON).click()
